Python: Getting variable from a function that generates random numbers by pressing a button